package i18n

import (
	"strings"
)

// GetMessage 获取指定语言的消息
func GetMessage(key string, lang Language) string {
	msg, exists := Messages[key]
	if !exists {
		return key
	}

	switch strings.ToLower(string(lang)) {
	case string(CN):
		return msg.CN
	case string(EN):
		return msg.EN
	case string(VN):
		return msg.VN
	default:
		return msg.EN // 默认返回英文
	}
}

// GetMessages 获取所有语言的消息
func GetMessages(key string) Message {
	msg, exists := Messages[key]
	if !exists {
		return Message{
			CN: key,
			EN: key,
			VN: key,
		}
	}
	return msg
}
